package cn.wolfcode.service.impl;

import cn.wolfcode.domain.Permission;
import cn.wolfcode.mapper.PermissionMapper;
import cn.wolfcode.qo.PageResult;
import cn.wolfcode.qo.QueryObject;
import cn.wolfcode.service.IPermissionService;
import cn.wolfcode.util.RequiredPermission;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.aop.framework.AopProxyUtils;
import org.springframework.aop.support.AopUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Controller;
import org.springframework.stereotype.Service;

import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.util.Collections;
import java.util.List;
import java.util.Map;

@Service
public class PermissionServiceImpl implements IPermissionService{
    
    @Autowired
    private PermissionMapper permissionMapper;

    @Override
    public void deleteById(Long id) {

        permissionMapper.deleteById(id);
    }

    @Override
    public void insert(Permission permission) {
        permissionMapper.insert(permission);
    }

    @Override
    public Permission selectById(Long id) {
        return permissionMapper.selectById(id);
    }

    @Override
    public List<Permission> selectAll() {
        return permissionMapper.selectAll();
    }

    @Override
    public void updateById(Permission permission) {
        permissionMapper.updateById(permission);
    }

    @Override
    public PageInfo<Permission> query(QueryObject qo) {
        PageHelper.startPage(qo.getCurrentPage(), qo.getPageSize());

        //  查询满足条件的每页显示的数据
        List<Permission> permissions = permissionMapper.queryForList(qo);
        //  返回一个结果集
        return new PageInfo<>(permissions);
    }

    @Autowired
    private ApplicationContext ctx;

    @Override
    public void reload() {
        //  获取数据库中所有的权限表达式
        List<String> strings = permissionMapper.selectAllExpression();
        //  获取每一个控制器对象
        Map<String, Object> beansWithAnnotation = ctx.getBeansWithAnnotation(Controller.class);
        //  遍历当前的控制器对象
        for (Object clazz : beansWithAnnotation.values()) {
            //  如果获取的控制器对象
            //  不是其Cglib代理对象
            //  就相当于方法中没贴权限注解
            //  不再执行后面操作
            if (!AopUtils.isCglibProxy(clazz)) {
                continue;
            }
            System.out.println("==========");
            //  获取字节码对象(父类)
            //  shiro会帮其控制器进行cglib动态代理
            //  该代理实现方式是继承
            //  所以需要获取代理对象的父类
            //  才能获取控制器上的权限表达式和权限名称
            Class<?> aClass = clazz.getClass().getSuperclass();
            //  通过字节码对象拿到所有方法
            Method[] declaredMethods = aClass.getDeclaredMethods();
            //  遍历方法
            for (Method declaredMethod : declaredMethods) {
                //  判断方法上是否有贴自定义注解
                //  传入shiro中的RequiresPermissions
                RequiresPermissions annotation = declaredMethod.getAnnotation(RequiresPermissions.class);
                //  若获取的注解对象不为空
                if (annotation != null) {
                    //  获取权限名称
                    String name = annotation.value()[1];
                    //  获取权限表达式
                    String expression = annotation.value()[0];
                    //  判断数据库中的权限表达式和当前获取的权限表达式是否存在
                    //  若不存在则执行下面操作
                    if (!strings.contains(expression)) {
                        //  创建权限对象，并封装
                        Permission permission = new Permission();
                        //  设置上面获取的权限名称
                        permission.setName(name);
                        //  设置上面获取的权限表达式
                        permission.setExpression(expression);
                        //  将权限对象插入到数据库中
                        permissionMapper.insert(permission);
                    }
                }
            }
        }
    }

    @Override
    public List<String> selectByEmpId(Long id) {
        return permissionMapper.selectByEmpId(id);
    }
}
